package com.cwz.core.utils.jwt;

import com.cwz.core.constant.SecurityConstants;
import com.cwz.core.text.Convert;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

import java.security.PrivateKey;
import java.security.PublicKey;
import java.util.Map;

/**
 * @program: w-demo
 * @description: Jwt 工具类（RSA加密）
 * @author: Wen
 **/
public class JwtRsaUtils {
	/**
	 * 从数据声明生成令牌
	 *
	 * @param claims 		数据声明
	 * @param privateKey 	私钥
	 * @return 令牌
	 */
	public static String createToken(Map<String, Object> claims, PrivateKey privateKey) {
		return Jwts.builder().setClaims(claims).signWith(SignatureAlgorithm.RS256, privateKey).compact();
	}

	/**
	 * 从令牌中获取数据声明
	 *
	 * @param token 		令牌
	 * @param publicKey 	公钥
	 * @return 数据声明
	 */
	public static Claims parseToken(String token, PublicKey publicKey) {
		return Jwts.parser().setSigningKey(publicKey).parseClaimsJws(token).getBody();
	}

	/**
	 * 根据令牌获取用户标识
	 *
	 * @param token 		令牌
	 * @param publicKey 	公钥
	 * @return 用户ID
	 */
	public static String getUserKey(String token, PublicKey publicKey) {
		Claims claims = parseToken(token, publicKey);
		return getValue(claims, SecurityConstants.USER_KEY);
	}

	/**
	 * 根据令牌获取用户标识
	 *
	 * @param claims 身份信息
	 * @return 用户ID
	 */
	public static String getUserKey(Claims claims) {
		return getValue(claims, SecurityConstants.USER_KEY);
	}

	/**
	 * 根据令牌获取用户ID
	 *
	 * @param token 		令牌
	 * @param publicKey 	公钥
	 * @return 用户ID
	 */
	public static String getUserId(String token, PublicKey publicKey) {
		Claims claims = parseToken(token, publicKey);
		return getValue(claims, SecurityConstants.DETAILS_USER_ID);
	}

	/**
	 * 根据身份信息获取用户ID
	 *
	 * @param claims 身份信息
	 * @return 用户ID
	 */
	public static String getUserId(Claims claims) {
		return getValue(claims, SecurityConstants.DETAILS_USER_ID);
	}

	/**
	 * 根据令牌获取用户名
	 *
	 * @param token 		令牌
	 * @param publicKey 	公钥
	 * @return 用户名
	 */
	public static String getUserName(String token, PublicKey publicKey) {
		Claims claims = parseToken(token, publicKey);
		return getValue(claims, SecurityConstants.DETAILS_USERNAME);
	}

	/**
	 * 根据身份信息获取用户名
	 *
	 * @param claims 身份信息
	 * @return 用户名
	 */
	public static String getUserName(Claims claims) {
		return getValue(claims, SecurityConstants.DETAILS_USERNAME);
	}

	/**
	 * 根据令牌获取用户公司
	 *
	 * @param token 		令牌
	 * @param publicKey 	公钥
	 * @return 用户公司
	 */
	public static String getCompany(String token, PublicKey publicKey) {
		Claims claims = parseToken(token, publicKey);
		return getValue(claims, SecurityConstants.DETAILS_COMPANY);
	}

	/**
	 * 根据身份信息获取用户公司
	 *
	 * @param claims 身份信息
	 * @return 用户公司
	 */
	public static String getCompany(Claims claims) {
		return getValue(claims, SecurityConstants.DETAILS_COMPANY);
	}

	/**
	 * 根据身份信息获取键值
	 *
	 * @param claims 身份信息
	 * @param key    键
	 * @return 值
	 */
	public static String getValue(Claims claims, String key) {
		return Convert.toStr(claims.get(key), "");
	}
}
